# Databricks notebook source
# MAGIC %md
# MAGIC # Installations

# COMMAND ----------

# MAGIC %pip install -U mostlyai[local] 
# MAGIC %pip install ipywidgets
# MAGIC %restart_python

# COMMAND ----------

import pandas as pd
from mostlyai.sdk import MostlyAI

mostly = MostlyAI(local=True)

# COMMAND ----------

# MAGIC %md
# MAGIC # Get synthetic data

# COMMAND ----------

df_samples = mostly.synthetic_datasets.get("bbd81e54-1833-4bb8-964c-d96480a6821e")

# COMMAND ----------

df_samples.data()

# COMMAND ----------

# MAGIC %md
# MAGIC # Save synthetic data

# COMMAND ----------

df_solicitud = spark.createDataFrame(df_samples.data()['solicitud'])
df_interopok = spark.createDataFrame(df_samples.data()['interopok'])
df_interopko = spark.createDataFrame(df_samples.data()['interopko'])
df_fullquery = spark.createDataFrame(df_samples.data()['fullquery'])
df_titular = spark.createDataFrame(df_samples.data()['titular'])
df_pagamentnomina = spark.createDataFrame(df_samples.data()['pagamentnomina'])
df_esborrany = spark.createDataFrame(df_samples.data()['esborrany'])
df_relationship = spark.createDataFrame(df_samples.data()['relationship'])
df_actuacionsnomina = spark.createDataFrame(df_samples.data()['actuacionsnomina'])

# COMMAND ----------

df_solicitud.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_solicitud_01_04")
df_interopok.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_interopok_01_04")
df_interopko.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_interopko_01_04")
df_fullquery.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_fullquery_01_04")
df_titular.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_titular_01_04")
df_pagamentnomina.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_pagamentnomina_01_04")
df_esborrany.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_esborrany_01_04")
df_relationship.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_relationship_01_04")
df_actuacionsnomina.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_actuacionsnomina_01_04")